coding python for projectile simulation